package FloodFill;

import java.util.LinkedList;
import java.util.Queue;

public class FloodFill {
    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        int prev = image[sr][sc]; // 记录初始点颜色
        if(prev == color) {
            // 如果初始点颜色就和新颜色相同, 那还改个蛋, 直接返回
            return image;
        }

        // 辅助向量数组
        int[] dx = {1,-1,0,0};
        int[] dy = {0,0,-1,1};

        int n = image[0].length; // 对应 y 坐标, y 不能超过的范围
        int m = image.length; // 对应 x 坐标, x 不能超过的范围

        // 用队列保存符合要求的点
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(new int[]{sr, sc});
        // 像素点更新颜色(存进来的都是符合要求的所以可以直接改)
        image[sr][sc] = color;

        while(!queue.isEmpty()) {
            // 取出上一层的节点
            int[] point = queue.poll();

            // 分解点坐标
            int Ox = point[0];
            int Oy = point[1];

            // BFS遍历下一层节点, offer符合要求的点
            for(int i = 0; i < 4; i++) {
                int x = Ox + dx[i];
                int y = Oy + dy[i];

                if(x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev) {
                    queue.offer(new int[]{x, y});
                    image[x][y] = color;
                }
            }
        }

        return image;
    }
}
